package com.demo.server.config;

import java.util.List;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import com.alibaba.fastjson2.JSON;
import com.demo.server.constant.CodeMsg;
import com.demo.server.mapper.PermissionMapper;
import com.demo.server.util.Result;

import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.exception.NotLoginException;
import cn.dev33.satoken.filter.SaServletFilter;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;

@Configuration
public class SaTokenConfigure {

	@Autowired
	private PermissionMapper permissionMapper;
	// 路径白名单
	static String[] excludes = {
//		"/**",// 取消认证
			"/user/login", // 登录
			"/h2-console",//h2数据库地址
			"/doc.html", // api地址
			"/webjars/**", // api资源
			"/swagger-ui/**", // swagger
			"/v3/**", // swagger
			"/favicon.ico", // swagger
			"/public/**", // 公共接口
			"/", // 公共接口
			"/login.html", 
			"/index.html", 
			"/**/*.js", // 公共接口
			"/**/*.css", // 公共接口
			"/**/*.jpg", // 公共接口
			"/**/*.png", // 公共接口
			"/**/*.woff", // 公共接口
			"/**/*.woff2", // 公共接口
			"/**/*.ttf", // 公共接口
			"/**/*.yml", // 公共接口
			"/**/*.json",// 公共接口
	};

	/**
	 * 注册 [Sa-Token全局过滤器]
	 */
	@Bean
	public SaServletFilter getSaServletFilter() {
		return new SaServletFilter()

				// 指定 拦截路由 与 放行路由
				.addInclude("/**").addExclude(excludes)

				// 认证函数: 每次请求执行
				.setAuth(obj -> {
					System.out.println("---------- 进入Sa-Token全局认证 -----------");
					String uri = SaHolder.getRequest().getRequestPath();
					if("GET".equals(SaHolder.getRequest().getMethod()) && !SaHolder.getRequest().isAjax()) {
						SaHolder.getResponse().redirect(("/index.html"));
						SaRouter.back();
					}
					System.out.println(uri);
					// 登录认证 -- 拦截所有路由，并排除/user/doLogin 用于开放登录
					// 非配置路径直接放行
					StpUtil.checkLogin();
					List<String> urls = permissionMapper.selectAllUrl();
					if (uri.startsWith("/")) {
						uri = uri.substring(1);
					}
					if (urls.contains(uri)) {
						StpUtil.checkPermission(uri);
					}
					SaRouter.stop();
					// 更多拦截处理方式，请参考“路由拦截式鉴权”章节
				})

				// 异常处理函数：每次认证函数发生异常时执行此函数

				.setError(e -> {
					System.out.println("---------- 进入Sa-Token异常处理 -----------" + e.getClass());
					// e.printStackTrace();
					if (e instanceof NotLoginException) {
						// SaResult.get(CodeMsg.LOGIN_PLEASE.getRetCode(),CodeMsg.LOGIN_PLEASE.getMessage(),
						// null);
						return JSON.toJSON(Result.result(CodeMsg.NOLOGIN));
					}
					// return
					// SaResult.get(CodeMsg.NO_RIGHT.getRetCode(),CodeMsg.NO_RIGHT.getMessage(),null
					// );
					return JSON.toJSON(Result.result(CodeMsg.NORIGHT));
				})

				// 前置函数：在每次认证函数之前执行
				.setBeforeAuth(r -> {
					// ---------- 设置一些安全响应头 ----------
					SaHolder.getResponse()
							// 服务器名称
							.setServer("sa-server")
							// 是否可以在iframe显示视图： DENY=不可以 | SAMEORIGIN=同域下可以 | ALLOW-FROM uri=指定域名下可以
							.setHeader("X-Frame-Options", "SAMEORIGIN")
							// 是否启用浏览器默认XSS防护： 0=禁用 | 1=启用 | 1; mode=block 启用, 并在检查到XSS攻击时，停止渲染页面
							.setHeader("X-XSS-Protection", "1; mode=block")
							// 禁用浏览器内容嗅探
							.setHeader("Content-Type", "application/json; charset=utf-8")
							// 禁用浏览器内容嗅探
							.setHeader("X-Content-Type-Options", "nosniff");
				});
	}

}